package com.point.travelaiagent.LLMStrategy;
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Component;

@Component("ollamaModelStrategy")
public class OllamaModelStrategy implements ChatModelStrategy {

    private final ChatModel ollamaChatModel;

    public OllamaModelStrategy(@Qualifier("ollamaChatModel") ChatModel ollamaChatModel) {
        this.ollamaChatModel = ollamaChatModel;
    }

    @Override
    public ChatModel getChatModel() {
        return ollamaChatModel;
    }

    @Override
    public String getModelType() {
        return "ollama";
    }
}
